import torch
print(torch.cuda.is_available())  # True 表示可以用GPU
print(torch.cuda.device_count())  # 设备数量
print(torch.cuda.current_device()) 
print(torch.cuda.get_device_name(0))